package com.opengate.base.redis;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.Cursor;
import org.springframework.data.redis.core.ScanOptions;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description Redis客户端 操作Redis数据
 * @date 2023/3/10 2:32 PM
 */
@Slf4j
@Component
public class RedisHelper {

    /** Redis客户端 */
    private StringRedisTemplate stringRedisTemplate;

    @Autowired
    public void setStringRedisTemplate(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }

    /**
     * 向redis中设值
     *
     * @param key 使用 a:b:id 的形式，便于管理
     * @param value 值
     * @return 是否设置成功
     */
    public boolean set(String key, String value) {
        boolean result = false;
        try {
            stringRedisTemplate.opsForValue().set(key, value);
            result = true;
        } catch (Exception ex) {
            log.error("Redis设置缓存失败，原因：{}", ex.getMessage());
        }

        return result;
    }

    /**
     * 向redis中设值，同时设置过期时间
     *
     * @param key 使用 a:b:id 的形式，便于管理
     * @param value 值
     * @param time 过期时间
     * @return 是否设置成功
     */
    public boolean set(String key, String value, Long time) {
        boolean result = false;
        try {
            stringRedisTemplate.opsForValue().set(key, value);
            expire(key, time);
            result = true;
        } catch (Exception ex) {
            log.error("Redis设置过期缓存失败，原因：{}", ex.getMessage());
        }

        return result;
    }

    /**
     * 获取redis中的值
     *
     * @param key redis键
     * @return 值
     */
    public String get(String key) {
        String result = null;
        try {
            result = stringRedisTemplate.opsForValue().get(key);
        } catch (Exception ex) {
            log.error("Redis获取缓存失败，原因：{}", ex.getMessage());
        }

        return result;
    }

    /**
     * 设置key的过期时间 单位：s
     *
     * @param key 键
     * @param time 时间
     * @return 是否设置成功
     */
    public boolean expire(String key, Long time) {
        boolean result = false;
        try {
            if (time > 0) {
                stringRedisTemplate.expire(key, time, TimeUnit.SECONDS);
                result = true;
            }
        } catch (Exception ex) {
            log.error("Redis设置缓存过期时间失败，原因：{}", ex.getMessage());
        }

        return result;
    }

    /**
     * 根据key删除对应value
     *
     * @param key 键
     * @return 是否删除成功
     */
    public boolean remove(String key) {
        boolean result = false;
        try {
            stringRedisTemplate.delete(key);
            result = true;
        } catch (Exception ex) {
            log.error("Redis移除缓存失败，原因：{}", ex.getMessage());
        }

        return result;
    }

    /**
     * 根据key判断缓存中是否有对应的value
     *
     * @param key 键
     * @return 是否存在
     */
    public boolean exists(String key) {
        boolean result = false;
        try {
            result = Boolean.TRUE.equals(stringRedisTemplate.hasKey(key));
        } catch (Exception ex) {
            log.error("Redis查询缓存是否存在失败，原因：{}", ex.getMessage());
        }

        return result;
    }

    /**
     * 根据表达式批量删除Keys 异步方法
     *
     * <p>如果需要扫描大量key，建议使用Redis的SCAN命令来逐步遍历key。
     *
     * @param pattern 表达式
     */
    @Async
    public void removeBatchByPatternKeys(String pattern) {
        try {
            Set<String> keys = stringRedisTemplate.keys(pattern);
            if (CollectionUtils.isNotEmpty(keys)) {
                stringRedisTemplate.delete(keys);
            }
        } catch (Exception ex) {
            log.error("Redis批量删除Keys失败，原因：{}", ex.getMessage());
        }
    }

    /**
     * 根据表达式批量删除Keys 异步方法
     *
     * <p>SCAN方式实现
     *
     * <p>推荐
     *
     * @param pattern 表达式
     */
    @Async
    public void removeBatchByPatternScan(String pattern) {
        try {
            List<String> keys = keys(pattern);
            if (CollectionUtils.isNotEmpty(keys)) {
                stringRedisTemplate.delete(keys);
            }
        } catch (Exception ex) {
            log.error("Redis批量删除Keys失败，原因：{}", ex.getMessage());
        }
    }

    /**
     * 获取符合条件的key
     *
     * @param pattern 表达式
     * @return 符合条件的key数组
     */
    public List<String> keys(String pattern) {
        List<String> keys = new ArrayList<>();
        this.scan(
                pattern,
                item -> {
                    // 符合条件的key
                    String key = new String(item, StandardCharsets.UTF_8);
                    keys.add(key);
                });

        return keys;
    }

    /**
     * 扫描符合条件的key
     *
     * @param pattern 表达式
     * @param consumer 消费者
     */
    public void scan(String pattern, Consumer<byte[]> consumer) {
        stringRedisTemplate.execute(
                (RedisConnection connection) -> {
                    try (Cursor<byte[]> cursor =
                            connection.scan(
                                    ScanOptions.scanOptions()
                                            .count(Long.MAX_VALUE)
                                            .match(pattern)
                                            .build())) {
                        cursor.forEachRemaining(consumer);

                        return null;
                    } catch (IOException e) {
                        log.error("Redis扫描符合条件的key失败，原因：{}", e.getMessage());

                        return null;
                    }
                });
    }
}
